from ProcessFile import ProcessFile
from ChatBot import Chatbot
# 文本拆分函数->将故事拆分为num段，并存储在txt中（json格式）

# 读取文本（根据故事名），并发送给llm，将收到的信息整理为合格prompt，或json存储

# 解析prompt json发送给LLM





if __name__ == '__main__':
    # 生成需要处理的文件名
    n = 2
    files = [f'story{i}' for i in range(1, n+1)] 
    # 预处理去重
    ps = ProcessFile(files)
    ps.process()
    # 加载QA模型对文本内容拓展
    api_url = "http://localhost:30014/v1/chat/completions"
    model_name = "Yi-1.5-9B-Chat"
    init_prompt = './prompt/QA_model_init_prompt.txt' # 初始化辅助QA模型
    chatbot = Chatbot(api_url, model_name)
    chatbot.generate_prompt(files, init_prompt)

